How to draw graphics in python